package com.atguigu.rabbitmq.springbootrabbitmq.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 *
 * @author 连志阳
 * @since 2022/2/10
 */
@Slf4j
@Component//实例化
public class MyCallBack implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnsCallback {
    
    @Autowired
    private RabbitTemplate rabbitTemplate;
    
    @PostConstruct
    public void init(){
        //注入
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnsCallback(this);
    }
    
    /**
     * 交换机确认回调方法
     * 1.发消息，交换机接收到，回调
     *   1.1 correlationData 保存回调消息的ID及相关信息
     *   1.2 交换机收到信息 ack = true
     *   1.3 cause null
     * 2.发消息，交换机接收失败，回调
     *   2.1 correlationData 保存回调消息的ID及相关信息
     *   2.2 交换机收到消息 ack = false
     *   2.3 cause 失败原因
     * @param correlationData 保存回调消息的ID及相关信息
     * @param ack 交换机是否收到消息
     * @param cause 失败原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        String id = correlationData != null ? correlationData.getId() : "";
        if(ack){
            log.info("交换机已经收到 id 为:{}的消息",id);
        }else{
            log.info("交换机还未收到 id 为:{}消息,由于原因:{}",id,cause);
        }
    }
    
    /**
     * 可以在当消息传递过程中不可达目的地时将消息返回给生产者
     * 只有不可达目的地的时候，才进行回退
     * @param returned 几个参数的封装对象
     */
    @Override
    public void returnedMessage(ReturnedMessage returned) {
        log.error("消息{}，被交换机{}退回，退回原因：{}，路由Key：{}",
                    new String(returned.getMessage().getBody()),
                        returned.getExchange(),
                        returned.getReplyText(),
                        returned.getRoutingKey());
    }
    
}
